package com.gqh.weather.controller;

import com.gqh.weather.exception.Result;
import com.gqh.weather.util.MinioUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.*;

/**
 * 文件上传控制器
 */
@Slf4j
@RestController
@RequestMapping("/pingjia/zh/v1/file/upload")
public class FileController {

    @Autowired
    private MinioUtil minioUtil;

    /**
     * 图片文件上传
     *
     * @param file 上传的文件
     * @return 上传结果，包含文件URL
     */
    @PostMapping("/picture")
    public Result<Map<String, String>> uploadPicture(@RequestParam("file") MultipartFile file) {
        log.info("图片文件上传，文件名: {}", file.getOriginalFilename());
        try {
            // 检查文件格式
            String originalFilename = file.getOriginalFilename();
            if (originalFilename == null || originalFilename.isEmpty()) {
                return Result.error("文件名不能为空");
            }

            String ext = originalFilename.substring(originalFilename.lastIndexOf(".")).toLowerCase();
            List<String> allowedExtensions = Arrays.asList(".jpg", ".jpeg", ".png");
            if (!allowedExtensions.contains(ext)) {
                return Result.error("图片格式不支持，请上传PNG/JPEG/JPG格式");
            }

            // 上传到MinIO
            String fileUrl = minioUtil.savePictureFile(file);
            
            // 构造返回结果
            Map<String, String> result = new HashMap<>();
            result.put("url", fileUrl);
            
            return Result.success(result);
        } catch (IOException e) {
            log.error("图片文件上传失败", e);
            return Result.error("文件上传失败：" + e.getMessage());
        } catch (Exception e) {
            log.error("图片文件上传失败", e);
            return Result.error("服务器错误：" + e.getMessage());
        }
    }

    /**
     * Excel文件上传
     *
     * @param file 上传的文件
     * @return 上传结果，包含文件URL
     */
    @PostMapping("/excel")
    public Result<Map<String, String>> uploadExcel(@RequestParam("file") MultipartFile file) {
        log.info("Excel文件上传，文件名: {}", file.getOriginalFilename());
        try {
            // 检查文件格式
            String originalFilename = file.getOriginalFilename();
            if (originalFilename == null || originalFilename.isEmpty()) {
                return Result.error("文件名不能为空");
            }

            String ext = originalFilename.substring(originalFilename.lastIndexOf(".")).toLowerCase();
            if (!".xlsx".equals(ext)) {
                return Result.error("Excel格式不支持，请上传xlsx格式");
            }

            // 上传到MinIO
            String fileUrl = minioUtil.saveExcelFile(file);
            
            // 构造返回结果
            Map<String, String> result = new HashMap<>();
            result.put("url", fileUrl);
            
            return Result.success(result);
        } catch (IOException e) {
            log.error("Excel文件上传失败", e);
            return Result.error("文件上传失败：" + e.getMessage());
        } catch (Exception e) {
            log.error("Excel文件上传失败", e);
            return Result.error("服务器错误：" + e.getMessage());
        }
    }

    /**
     * PDF文件上传
     *
     * @param file 上传的文件
     * @return 上传结果，包含文件URL
     */
    @PostMapping("/pdf")
    public Result<Map<String, String>> uploadPdf(@RequestParam("file") MultipartFile file) {
        log.info("PDF文件上传，文件名: {}", file.getOriginalFilename());
        try {
            // 检查文件格式
            String originalFilename = file.getOriginalFilename();
            if (originalFilename == null || originalFilename.isEmpty()) {
                return Result.error("文件名不能为空");
            }

            String ext = originalFilename.substring(originalFilename.lastIndexOf(".")).toLowerCase();
            if (!".pdf".equals(ext)) {
                return Result.error("文件格式不支持，请上传pdf格式");
            }

            // 上传到MinIO
            String fileUrl = minioUtil.savePdfFile(file);
            
            // 构造返回结果
            Map<String, String> result = new HashMap<>();
            result.put("url", fileUrl);
            
            return Result.success(result);
        } catch (IOException e) {
            log.error("PDF文件上传失败", e);
            return Result.error("文件上传失败：" + e.getMessage());
        } catch (Exception e) {
            log.error("PDF文件上传失败", e);
            return Result.error("服务器错误：" + e.getMessage());
        }
    }
}